#ifndef ML_DECISION_STUMP_PARAMS_H_
#define ML_DECISION_STUMP_PARAMS_H_

#include "WeakLearner.h"
#include <set>

GLOBAL_NAMESPACE_BEGIN
NAMESPACE_MACHINE_LEARNING_BEGIN

class MLDecisionStumpParams
{
public:
    //std::set<int> subClass;
    std::vector<int> subClass;
    int              dimF;
    bool             bGreater;
    double           threshold;
    double           a;
    double           b;
    std::vector<double> k;  // size: maxClass

    int     threIdx;
    double wcP, wcN;
};

NAMESPACE_MACHINE_LEARNING_END
GLOBAL_NAMESPACE_END

#endif//ML_DECISION_STUMP_PARAMS_H_